Tools Are Not Islands: Set-Level Tool Retrieval for LLM Agents via Query-Conditioned Hyperedge Prediction
工具并非孤岛:基于超边预测的LLM Agent集合级工具检索
🧭 Agent工具检索的集合级重定义与超图建模 ★ 深度解读
Shanghai Jiao Tong University · The Hong Kong Polytechnic University
将工具检索重构为查询条件超边预测,以工具集合整体为单位评分,并捕捉基数依赖的工具交互
📌 工作参考价值(4 点)
- 把工具集当作整体来打分:在电商 Agent 场景(如比价+下单+物流查询)中,不要只按单个 API 相关性召回,可参考 HYSET 的超边预测思想,对候选工具子集做联合评分,直接优化「工具集合是否完整覆盖任务」指标。
- 用基数特异性交互矩阵捕捉工具间依赖:多工具任务中,工具之间的互补性随集合规模变化。可以用一组按集合大小索引的交互矩阵(即 M2, M3, M4 …)来建模这种动态依赖,参数开销远小于显式高阶张量。
- 执行反馈可作为弱监督信号:即使只有部分标注数据,也可以在线采集工具调用链的最终任务成功与否作为奖励,通过 reward-weighted self-training 持续优化检索模块,尤其适合随时新增 API 的电商工具库。
- 两阶段推理兼顾效率与集合评分:先用轻量双编码器做 top-K 初筛,再用集合重排序在缩减后的空间里显式对比不同大小、不同组合的候选工具集,推理延迟可控(12 ms/query),可直接嵌入生产级 Agent 管线。